//
//  Created by 姚旭 on 2021/12/17.
//

import UIKit

public class CollectionViewFlowLayout: UICollectionViewFlowLayout {

    /// item size 获取方法作为一个属性
    /// 可以实现动态获取 item size
    /// 因此只需要配合原有属性设置即可快速实现一个 FlowLayout 样式集合视图
    public var itemSizeReader: ((UICollectionView) -> CGSize)?
    
}

public extension CollectionViewFlowLayout {
    
    override func prepare() {
        super.prepare()
        if let itemSizeReader, let collectionView {
            itemSize = itemSizeReader(collectionView)
        }
    }
    
}
